Searched defs:values_shape (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 33 def _has_valid_dims(weights_shape, values_shape): argument
47 weights_rank, weights_shape, values_rank, values_shape): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense_test.cc | 33 std::initializer_list<int> values_shape, T default_value, in SparseToDenseOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_sparse_kernel.cc | 189 const TensorShape& values_shape) { in MakeIndexSuffixes()
|
| D | ragged_gather_op.cc | 235 TensorShape values_shape = params_dense_values_in.shape(); in WriteValues() local
|
| D | example_parsing_ops.cc | 696 TensorShape values_shape({0}); in Compute() local
806 TensorShape values_shape({total_num_features}); in Compute() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1101 TensorShape values_shape; in FastParseExample() local
1176 TensorShape values_shape; in FastParseExample() local
1271 TensorShape values_shape; in FastParseSingleExample() local
1484 TensorShape values_shape; in FastParseSingleExample() local
2118 TensorShape indices_shape, values_shape; in FastParseSequenceExample() local
2316 TensorShape indices_shape, values_shape; in FastParseSequenceExample() local
|
| D | example_proto_helper.h | 87 TensorShape values_shape; member
|
| D | example_proto_helper.cc | 379 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; in BatchExampleProtoToTensors() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 672 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
|
| D | tuple_points_to_analysis_test.cc | 1090 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| D | hlo_dataflow_analysis_test.cc | 2469 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 2821 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1495 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()
|