| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 34 def _has_valid_dims(weights_shape, values_shape): argument
48 weights_rank, weights_shape, values_rank, values_shape): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lower_upper_bound_ops.cc | 36 const TensorShape values_shape = ctx->InputShape("values"); in BuildLowerUpperBoundOp() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_sparse_kernel.cc | 191 const TensorShape& values_shape) { in MakeIndexSuffixes()
|
| D | ragged_tensor_to_variant_op.cc | 52 TensorShape values_shape = batched_ragged.values().shape(); in UnbatchRaggedZerothDim() local
|
| D | ragged_gather_op.cc | 240 TensorShape values_shape = params_dense_values_in.shape(); in WriteValues() local
|
| D | example_parsing_ops.cc | 1052 TensorShape values_shape({0}); in Compute() local
1162 TensorShape values_shape({total_num_features}); in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense_test.cc | 34 std::initializer_list<int> values_shape, T default_value, in SparseToDenseOpModel()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite_test.cc | 42 std::initializer_list<int> values_shape, in RaggedTensorToTensorOpModel()
|
| D | ragged_tensor_to_tensor_tflite.cc | 254 RuntimeShape values_shape(values.dims->size, values.dims->data); in CalculateOutputSize() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1264 TensorShape values_shape; in FastParseExample() local
1322 TensorShape values_shape; in FastParseExample() local
1375 TensorShape values_shape; in FastParseExample() local
1482 TensorShape values_shape; in FastParseSingleExample() local
2350 TensorShape indices_shape, values_shape; in ParseContextSparseFeatures() local
2411 TensorShape values_shape, splits_shape; in ParseContextRaggedFeatures() local
2641 TensorShape indices_shape, values_shape; in ParseSequenceSparseFeatures() local
2769 TensorShape values_shape, inner_splits_shape, outer_splits_shape; in ParseSequenceRaggedFeatures() local
|
| D | example_proto_helper.h | 88 TensorShape values_shape; member
|
| D | example_proto_helper.cc | 382 const TensorShape& values_shape = sparse_tensor_batch_shapes.values_shape; in BatchExampleProtoToTensors() local
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 553 tf::TensorShape values_shape; in FastParseExampleLite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 1004 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
|
| D | hlo_dataflow_analysis_test.cc | 2586 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 3581 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2228 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()
|