Home
last modified time | relevance | path

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

1234

/external/ComputeLibrary/src/core/
DTensorInfo.cpp70 TensorInfo::TensorInfo(const TensorShape &tensor_shape, Format format) in TensorInfo()
82 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in TensorInfo()
88 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Qu… in TensorInfo()
95 TensorInfo::TensorInfo(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, Da… in TensorInfo()
113 void TensorInfo::init(const TensorShape &tensor_shape, Format format) in init()
123 void TensorInfo::init(const TensorShape &tensor_shape, Format format, in init()
140 void TensorInfo::init(const TensorShape &tensor_shape, size_t num_channels, DataType data_type) in init()
151 void TensorInfo::init(const TensorShape &tensor_shape, size_t num_channels, DataType data_type, in init()
182 size_t TensorInfo::init_auto_padding(const TensorShape &tensor_shape, Format format) in init_auto_padding()
193 size_t TensorInfo::init_auto_padding(const TensorShape &tensor_shape, size_t num_channels, DataType… in init_auto_padding()
DPyramidInfo.cpp44 PyramidInfo::PyramidInfo(size_t num_levels, float scale, const TensorShape &tensor_shape, Format fo… in PyramidInfo()
55 void PyramidInfo::init(size_t num_levels, float scale, const TensorShape &tensor_shape, Format form… in init()
92 const TensorShape &PyramidInfo::tensor_shape() const in tensor_shape() function in PyramidInfo
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()
145 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, in TensorShapeToXLAShape()
154 const TensorShape& tensor_shape) { in TensorShapeToXLAShape()
/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/tensorflow/tensorflow/core/data/
Ddataset.proto16 .tensorflow.TensorShapeProto tensor_shape = 2; field
/external/ComputeLibrary/src/runtime/
DSubTensor.cpp35 SubTensor::SubTensor(ITensor *parent, const TensorShape &tensor_shape, const Coordinates &coords, b… in SubTensor()
DPyramid.cpp55 TensorShape tensor_shape = _info.tensor_shape(); in internal_init() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dquantization_util.cc34 const RuntimeShape &tensor_shape, int32_t zero_point, in DequantizeInt8()
Dquantization_util_test.cc50 RuntimeShape tensor_shape(1, quantized_data.size()); in TEST() local
/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()
DCLPyramid.cpp61 TensorShape tensor_shape = _info.tensor_shape(); in internal_init() local
/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, 1U), DataLayout::NC… in do_setup() local
Dgraph_vgg16.cpp70 …const TensorShape tensor_shape = permute_shape(TensorShape(224U, 224U, 3U, 1U), DataLayout::NC… in do_setup() local
Dgraph_alexnet.cpp77 …const TensorShape tensor_shape = permute_shape(TensorShape(227U, 227U, 3U, 1U), DataLayout::NC… in do_setup() local
Dgraph_mnist.cpp74 …const TensorShape tensor_shape = permute_shape(TensorShape(28U, 28U, 1U), DataLayout::NCHW, op… in do_setup() local
Dgraph_srcnn955.cpp84 …const TensorShape tensor_shape = permute_shape(TensorShape(image_width, image_height, 3U, 1U),… in do_setup() local
/external/tensorflow/tensorflow/core/framework/
Dtensor.proto20 TensorShapeProto tensor_shape = 2; field
/external/tensorflow/tensorflow/core/ops/
Dlist_ops.cc182 shape_inference::ShapeHandle tensor_shape = c->UnknownShape(); in __anond381a6db0502() local
340 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anond381a6db0902() local
376 shape_inference::ShapeHandle tensor_shape = c->input(0); in __anond381a6db0a02() 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

1234