| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/ |
| D | tensor.pb.cc | 82 TensorProto::_Internal::tensor_shape(const TensorProto* msg) { in tensor_shape() function in tensorflow::TensorProto::_Internal
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | parameterized_truncated_normal_op.cc | 634 TensorShape tensor_shape; in Compute() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_ssd_mobilenet.cpp | 78 …const TensorShape tensor_shape = permute_shape(TensorShape(300, 300, 3U, 1U), DataLayout::NCHW… in do_setup() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | rebatch_dataset_op.cc | 485 TensorShape tensor_shape(dim_sizes); in GetNextInternal() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupWrapper.cpp | 151 for (const auto& tensor_shape : tensors_to_verify) { in verify_tensors() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | tensor_list_ops.cc | 588 const TensorShape& tensor_shape = ctx->InputShape(0); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 57 auto tensor_shape = RankedTensorType::get(shape, element_type); in CreateElementsAttrOfTypeValues() local
69 auto tensor_shape = RankedTensorType::get(shape, element_type); in CreateElementsAttrOfTypeValues() local
82 auto tensor_shape = RankedTensorType::get(shape, element_type); in CreateElementsAttrOfTypeValues() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | ConvolutionLayerFixture.h | 429 const TensorShape tensor_shape = tensor_info.tensor_shape(); in prepare_weights() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | graph_utils.cc | 90 std::unique_ptr<tensorflow::TensorShapeProto> tensor_shape = in AddScalarConstNodeHelper() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/ |
| H A D | util.cpp | 231 const auto& tensor_shape = in variantShapesToStr() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | pywrap_tensor.cc | 439 PyObject* tensor_shape; member
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | meta_graph.pb.h | 4777 inline const ::tensorflow::TensorShapeProto& TensorInfo::tensor_shape() const { in tensor_shape() function
4782 ::tensorflow::TensorShapeProto* tensor_shape) { in unsafe_arena_set_allocated_tensor_shape()
4824 inline void TensorInfo::set_allocated_tensor_shape(::tensorflow::TensorShapeProto* tensor_shape) { in set_allocated_tensor_shape()
|
| D | meta_graph.pb.cc | 3332 TensorInfo::_Internal::tensor_shape(const TensorInfo* msg) { in tensor_shape() function in tensorflow::TensorInfo::_Internal
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_io_utils.cc | 65 const TensorShapeProto& tensor_shape) { in PrepareChunkEventProto()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | export_tensorflow.cc | 289 auto* tensor_shape = tensor->mutable_tensor_shape(); in CreateIntTensorConst() local
374 auto* tensor_shape = tensor->mutable_tensor_shape(); in CreateReshapeShapeTensorConst() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | tensor_desc.cc | 359 const BHWDC& tensor_shape, GenericGPUResourcesWithValue* resources) const { in GetGpuResources() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesNorm.cpp | 446 const auto tensor_shape = tensor.sizes(); in has_same_shape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | tensor.cc | 1187 const TensorShape& tensor_shape, const char* data, in SummarizeArray()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIterator.cpp | 588 auto tensor_shape = invert_perm(shape_); in allocate_or_resize_outputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | c_api.cc | 1023 const auto& tensor_shape = default_value.shape(); in SetOpAttrValueScalar() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_function_test.cc | 1599 TF_Tensor* tensor_shape = Int32Tensor({37, 1}); in DefineStatefulFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantize_model.cc | 631 const std::vector<int32_t>& tensor_shape = tensor->shape; in RequantizeConstant() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device.cc | 1927 std::vector<int64_t> tensor_shape(TensorShapeAsVector(input, status)); in Execute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 1545 auto tensor_shape = in CalculateTensorElementCount() local
|
| H A D | op_level_cost_estimator_test.cc | 1341 TensorShape tensor_shape(tensor_shape_proto); in ExpectTensorShape() local
|