Home
last modified time | relevance | path

Searched refs:bhwc (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/
Dinterpreter_utils.cc71 TensorFloat32 bhwc; in InterpreterInvokeWithOpResolver() local
72 bhwc.id = t; in InterpreterInvokeWithOpResolver()
77 bhwc.shape.b = out_tensor->dims->data[0]; in InterpreterInvokeWithOpResolver()
80 bhwc.shape.h = 1; in InterpreterInvokeWithOpResolver()
81 bhwc.shape.w = 1; in InterpreterInvokeWithOpResolver()
82 bhwc.shape.c = out_tensor->dims->data[1]; in InterpreterInvokeWithOpResolver()
85 bhwc.shape.h = 1; in InterpreterInvokeWithOpResolver()
86 bhwc.shape.w = out_tensor->dims->data[1]; in InterpreterInvokeWithOpResolver()
87 bhwc.shape.c = out_tensor->dims->data[2]; in InterpreterInvokeWithOpResolver()
90 bhwc.shape.h = out_tensor->dims->data[1]; in InterpreterInvokeWithOpResolver()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dtranspose.cc73 const std::string bhwc[] = {batch_id, "Y", "X", "dst_channel"}; in GetTransposeCode() local
75 c += " args.src_tensor.SetBatchRef(" + bhwc[remap[0]] + ");\n"; in GetTransposeCode()
77 c += " int s_y = " + bhwc[remap[1]] + ";\n"; in GetTransposeCode()
78 c += " int s_x = " + bhwc[remap[2]] + ";\n"; in GetTransposeCode()
79 c += " int s_c = " + bhwc[remap[3]] + ";\n"; in GetTransposeCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_helper.cc96 absl::Status ExtractTensorShape(const TfLiteTensor& tflite_tensor, BHWC* bhwc) { in ExtractTensorShape() argument
101 *bhwc = BHWC(dims->data[0], 1, 1, 1); in ExtractTensorShape()
105 *bhwc = BHWC(dims->data[0], 1, 1, dims->data[1]); in ExtractTensorShape()
109 *bhwc = BHWC(dims->data[0], 1, dims->data[1], dims->data[2]); in ExtractTensorShape()
113 *bhwc = BHWC(dims->data[0], dims->data[1], dims->data[2], dims->data[3]); in ExtractTensorShape()
Dmodel_builder_helper.h46 absl::Status ExtractTensorShape(const TfLiteTensor& tflite_tensor, BHWC* bhwc);
Dmodel_builder.cc2009 auto read_bhwc = [&](int tensor_index, BHWC* bhwc) -> absl::Status { in ReadAttribsWithBatch() argument
2012 *bhwc = BHWC(t.data[0], t.data[1], t.data[2], t.data[3]); in ReadAttribsWithBatch()
2027 auto read_hwc = [&](int tensor_index, BHWC* bhwc) -> absl::Status { in ReadAttribsWithoutBatch() argument
2030 *bhwc = BHWC(0, t.data[0], t.data[1], t.data[2]); in ReadAttribsWithoutBatch()