Home
last modified time | relevance | path

Searched defs:full_shape (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dsvd_op_test.py407 full_shape = batch_dims + (rows, cols) variable
426 full_shape = batch_dims + mat_shape variable
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/signal/
H A Dreconstruction_ops.py77 def full_shape(inner_shape): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariable_scope.py60 def __init__(self, full_shape, var_offset): argument
101 def full_shape(self): member in _PartitionInfo
2668 def _iter_slices(full_shape, num_slices, slice_dim): argument
H A Dvariables.py1267 full_shape=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dvariable.proto79 repeated int64 full_shape = 2; field
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dvariable.pb.h989 inline ::int64_t SaveSliceInfoDef::full_shape(int index) const { in full_shape() function
1009 SaveSliceInfoDef::full_shape() const { in full_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc535 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
574 Shape full_shape = ShapeUtil::MakeShape(F32, {4, 100, 10, 100, 50}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
H A Dxla_sharding.py478 full_shape, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle.cc1011 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier_test.cc5046 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local