Searched defs:full_shape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | reconstruction_ops.py | 77 def full_shape(inner_shape): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | svd_op_test.py | 407 full_shape = batch_dims + (rows, cols) variable
426 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 | 478 full_shape, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 60 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
|
| D | variables.py | 1267 full_shape=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 535 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
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 1011 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 5046 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
|