Searched refs:tensor_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 71 auto tensor_dims = tensor.flat<int64>(); in ShapeTensorToTensorShape() local
74 dims[i] = tensor_dims(i); in ShapeTensorToTensorShape()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 164 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in FillPerChannelMinMax() local
173 const float val = input[Offset(tensor_dims, indices)]; in FillPerChannelMinMax()
377 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in SymmetricPerChannelQuantizeValues() local
384 int index = Offset(tensor_dims, indices); in SymmetricPerChannelQuantizeValues()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 448 uint32_t* tensor_dims = reinterpret_cast<uint32_t*>(tensor->dims->data); in ConvertTensorTypeToNNType() local
453 tensor_dims = &scalar_rank; in ConvertTensorTypeToNNType()
458 .dimensions = tensor_dims, in ConvertTensorTypeToNNType()
1113 const uint32_t* tensor_dims = reinterpret_cast<const uint32_t*>(dims->data); in AddNewInputConstantTensor() local
1114 ANeuralNetworksOperandType operand_type{nn_type, tensor_rank, tensor_dims, in AddNewInputConstantTensor()
1379 uint32_t* tensor_dims = in AddTensor() local
1386 tensor_dims = &tensor_rank; in AddTensor()
1390 tensor_dims = nullptr; in AddTensor()
1393 ANeuralNetworksOperandType operand_type{nn_type, tensor_rank, tensor_dims, in AddTensor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1361 int tensor_dims[] = {1, input_size, 1}; in Create() local
1362 int strides[] = {tensor_dims[1] * tensor_dims[2], tensor_dims[2], 1}; in Create()
1365 /*nbDims=*/sizeof(tensor_dims) / sizeof(tensor_dims[0]), in Create()
1366 /*dimA=*/tensor_dims, in Create()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2393 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F() local
2401 EXPECT_EQ(tensor_dims[i][j], TF_Dim(v, j)) in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4774 std::vector<int> tensor_dims(dims.d, dims.d + dims.nbDims); in ConvertPack() local
4775 tensor_dims.insert(tensor_dims.begin() + trt_axis, 1); in ConvertPack()
4777 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(tensor_dims, &expanded_dims)); in ConvertPack()
|
| D | convert_nodes_test.cc | 2359 std::vector<int> tensor_dims; in TEST_F() member
2373 const std::vector<int>& dims = params[i].tensor_dims; in TEST_F()
2397 AddTestTensor("input", ok_params[i].tensor_dims, batch_size); in TEST_F()
|