Home
last modified time | relevance | path

Searched defs:tensor_shape (Results 1 – 25 of 96) sorted by relevance

1234

/external/ComputeLibrary/src/core/
DTensorInfo.cpp93 TensorInfo::TensorInfo(const TensorShape &tensor_shape, Format format) in TensorInfo()
105 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in TensorInfo()
111 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Qu… in TensorInfo()
118 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Da… in TensorInfo()
130 void TensorInfo::init(const TensorShape &tensor_shape, Format format) in init()
140 void TensorInfo::init(const TensorShape &tensor_shape, Format format, in init()
157 void TensorInfo::init(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in init()
168 void TensorInfo::init(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, in init()
185 size_t TensorInfo::init_auto_padding(const TensorShape &tensor_shape, Format format) in init_auto_padding()
196 size_t TensorInfo::init_auto_padding(const TensorShape &tensor_shape, size_t num_channels, DataType… in init_auto_padding()
DSubTensorInfo.cpp63 SubTensorInfo::SubTensorInfo(ITensorInfo *parent, TensorShape tensor_shape, Coordinates coords, boo… in SubTensorInfo()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc89 TensorShape* tensor_shape) { in XLAShapeToTensorShape()
104 const PartialTensorShape& tensor_shape, in TensorShapeToXLAShape()
113 const PartialTensorShape& tensor_shape) { in TensorShapeToXLAShape()
137 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape()
146 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
153 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dquantization_util.cc35 const RuntimeShape &tensor_shape, int32_t zero_point, in DequantizeInt8()
46 const RuntimeShape& tensor_shape, in PerChannelDequantizeInt8()
Dquantization_util_test.cc50 RuntimeShape tensor_shape(1, quantized_data.size()); in TEST() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_param_op.cc34 PartialTensorShape tensor_shape; in XlaFakeParamOp() local
Dsendrecv_ops.cc68 TensorShape tensor_shape; in RecvOp() local
/external/ComputeLibrary/src/runtime/
DSubTensor.cpp35 SubTensor::SubTensor(ITensor *parent, const TensorShape &tensor_shape, const Coordinates &coords, b… in SubTensor()
/external/tensorflow/tensorflow/core/data/
Ddataset.proto18 .tensorflow.TensorShapeProto tensor_shape = 2; field
/external/tensorflow/tensorflow/python/keras/protobuf/
Dprojector_config.proto36 repeated uint32 tensor_shape = 4; field
/external/tensorflow/tensorflow/core/protobuf/
Dsnapshot.proto39 .tensorflow.TensorShapeProto tensor_shape = 2; field
/external/ComputeLibrary/tests/datasets/
DROIDataset.h93 void add_config(TensorShape tensor_shape, ROIPoolingLayerInfo info, TensorShape rois_shape) in add_config()
/external/ComputeLibrary/src/runtime/CL/
DCLSubTensor.cpp37 CLSubTensor::CLSubTensor(ICLTensor *parent, const TensorShape &tensor_shape, const Coordinates &coo… in CLSubTensor()
/external/ComputeLibrary/examples/
Dgraph_lenet.cpp70 …const TensorShape tensor_shape = permute_shape(TensorShape(28U, 28U, 1U, batches), DataLayout:… in do_setup() local
Dgraph_vgg19.cpp69 …const TensorShape tensor_shape = permute_shape(TensorShape(224U, 224U, 3U, common_params.batch… in do_setup() local
Dgraph_alexnet.cpp77 …const TensorShape tensor_shape = permute_shape(TensorShape(227U, 227U, 3U, common_params.batch… in do_setup() local
Dgraph_vgg16.cpp70 …const TensorShape tensor_shape = permute_shape(TensorShape(224U, 224U, 3U, common_params.batch… in do_setup() local
Dgraph_srcnn955.cpp84 …const TensorShape tensor_shape = permute_shape(TensorShape(image_width, image_height, 3U, comm… in do_setup() local
Dgraph_vgg_vdsr.cpp85 …const TensorShape tensor_shape = permute_shape(TensorShape(image_width, image_height, 1U, comm… in do_setup() local
Dgraph_squeezenet_v1_1.cpp70 …const TensorShape tensor_shape = permute_shape(TensorShape(227U, 227U, 3U, common_params.batch… in do_setup() local
Dgraph_squeezenet.cpp70 …const TensorShape tensor_shape = permute_shape(TensorShape(224U, 224U, 3U, common_params.batch… in do_setup() local
/external/tensorflow/tensorflow/core/framework/
Dtensor.proto20 TensorShapeProto tensor_shape = 2; field
/external/tensorflow/tensorflow/core/ops/
Dlist_ops.cc185 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anon398e29250502() local
344 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anon398e29250902() local
382 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anon398e29250a02() local
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc118 const RuntimeShape tensor_shape(tensor.dims->size, tensor.dims->data); in TensorShapeFromTensor() local
156 const RuntimeShape tensor_shape(tensor->dims->size, tensor->dims->data); in GetMaxWidthValueRowID() local
187 const RuntimeShape tensor_shape(tensor->dims->size, tensor->dims->data); in GetMaxWidthRowSplit() local
335 const RuntimeShape tensor_shape(value_rowids.dims->size, in CalculateOutputIndexValueRowID() local
/external/ComputeLibrary/tests/validation/CL/UNIT/
DMultithreaded.cpp57 const TensorShape tensor_shape(128u, 4u, 5u); in TEST_SUITE() local

1234