| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | global_pooling_to_reduce_op.cc | 34 bool IsGlobalPooling(const Pooling2DAttributes& attr, const BHWC& src_shape, in IsGlobalPooling()
43 const BHWC& src_shape, const BHWC& dst_shape) { in IsGlobalAveragePooling()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | cl_test.cc | 60 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
108 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | test_util.cc | 71 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 475 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce()
483 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
|
| D | conv_weights_converter_test_util.cc | 35 auto src_shape = in ConvolutionWeightsConverterTest() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 591 const TensorShape& src_shape, in ShapeFromFormat()
617 const TensorShape& src_shape, in ShapeFromFilterFormat()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.cc | 162 const BHWC& src_shape, in SelectReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 439 const Shape& src_shape) { in CopyElementsBetween()
1569 auto src_shape = src.subshape(); in CopyElementsWithDynamicBound() local
2377 const auto& src_shape = shape_->tuple_shapes(i); in MutableBorrowingLiteral() local
2436 const auto& src_shape = shape_->tuple_shapes(i); in BorrowingLiteral() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 441 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 466 const auto& src_shape = default_value_tensor.shape(); in SetOutput() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 119 const auto src_shape = inputs[i]->tensor.shape; in IsGenericAdd() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 99 const auto src_shape = inputs[i]->tensor.shape; in IsGenericAdd() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 994 const Shape& src_shape = ShapeUtil::GetTupleElementShape(operand_shape, 1); in HandleCopyDone() local
|
| D | algebraic_simplifier.cc | 239 const Shape& src_shape = operand_convert->operand(0)->shape(); in IsConvertPairNoOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 734 ArrayRef<int64_t> src_shape = in matchAndRewrite() local
|