| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 30 def _has_valid_dims(weights_shape, values_shape): argument
44 weights_rank, weights_shape, values_rank, values_shape): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_variant_op.cc | 41 TensorShape values_shape = batched_values.shape(); in UnbatchDenseZerothDim() local
92 TensorShape values_shape = batched_ragged.values().shape(); in UnbatchRaggedZerothDim() local
|
| D | ragged_tensor_to_sparse_kernel.cc | 202 const TensorShape& values_shape) { in MakeIndexSuffixes()
|
| D | ragged_gather_op.cc | 246 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/compiler/tf2xla/kernels/ |
| D | lower_upper_bound_ops.cc | 36 const TensorShape values_shape = ctx->InputShape("values"); in BuildLowerUpperBoundOp() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 1280 TensorShape values_shape; in FastParseExample() local
1338 TensorShape values_shape; in FastParseExample() local
1391 TensorShape values_shape; in FastParseExample() local
1498 TensorShape values_shape; in FastParseSingleExample() local
2366 TensorShape indices_shape, values_shape; in ParseContextSparseFeatures() local
2428 TensorShape values_shape, splits_shape; in ParseContextRaggedFeatures() local
2659 TensorShape indices_shape, values_shape; in ParseSequenceSparseFeatures() local
2787 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/ |
| D | sparse_to_dense_test.cc | 40 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/cc/gradients/ |
| D | array_grad.cc | 665 auto values_shape = Shape(scope, values); in BatchGatherGrad() local
730 auto values_shape = Concat(scope, {outer_shape, {-1}, inner_shape}, 0); in GatherV2Grad() local
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 555 tf::TensorShape values_shape; in FastParseExampleLite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 1175 Shape values_shape = ShapeUtil::MakeShape(F32, {2, 3}); in TEST_F() local
|
| D | hlo_dataflow_analysis_test.cc | 2647 Shape values_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 4017 Shape values_shape = ShapeUtil::MakeShape(S32, {5, 0}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2319 ShapeHandle values_shape, ShapeHandle shape_shape) { in ValidateSparseTensor()
|