Searched defs:full_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | reconstruction_ops.py | 81 def full_shape(inner_shape): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | svd_op_test.py | 377 full_shape = batch_dims + (rows, cols) variable
396 full_shape = batch_dims + mat_shape variable
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 79 repeated int64 full_shape = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ |
| D | xla_sharding.py | 389 def manual_to_auto_spmd_partition(tensor, manual_sharding, full_shape): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 68 def __init__(self, full_shape, var_offset): argument
111 def full_shape(self): member in _PartitionInfo
2572 def _iter_slices(full_shape, num_slices, slice_dim): argument
|
| D | variables.py | 1275 full_shape=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 553 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
592 Shape full_shape = ShapeUtil::MakeShape(F32, {4, 100, 10, 100, 50}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 998 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 4633 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
|