Searched defs:tensor_dims (Results 1 – 10 of 10) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_striding_sycl.cpp | 37 Eigen::array<IndexType, 4> tensor_dims = {{2,3,5,7}}; in test_simple_striding() local
119 Eigen::array<IndexType, 4> tensor_dims = {{2,3,5,7}}; in test_striding_as_lvalue() local
|
| D | cxx11_tensor_block_io.cpp | 222 DSizes<Index, 3> tensor_dims(7, 9, 7); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
269 DSizes<Index, 4> tensor_dims(7, 5, 9, 9); in test_block_io_copy_using_reordered_dimensions_squeeze() local
|
| D | cxx11_tensor_block_access.cpp | 73 DSizes<Index, 2> tensor_dims(100, 100); in test_block_mapper_sanity() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 71 auto tensor_dims = tensor.flat<int64>(); in ShapeTensorToTensorShape() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 165 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in FillPerChannelMinMax() local
378 RuntimeShape tensor_dims{dimension[0], dimension[1], dimension[2], in SymmetricPerChannelQuantizeValues() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 493 uint32_t* tensor_dims = reinterpret_cast<uint32_t*>(tensor->dims->data); in ConvertTensorTypeToNNType() local
1249 const uint32_t* tensor_dims = reinterpret_cast<const uint32_t*>(dims->data); in AddNewInputConstantTensor() local
1555 uint32_t* tensor_dims = in AddTensor() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2424 const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]}; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4368 std::vector<int> tensor_dims(dims.d, dims.d + dims.nbDims); in ConvertPack() local
|
| D | convert_nodes_test.cc | 2299 std::vector<int> tensor_dims; in TEST_P() member
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1466 int tensor_dims[] = {1, input_size, 1}; in Create() local
|