Searched defs:tensor_dims (Results 1 – 11 of 11) 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_t>(); in ShapeTensorToTensorShape() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 172 RuntimeShape tensor_dims = in FillPerChannelMinMax() local
388 RuntimeShape tensor_dims = in SymmetricPerChannelQuantizeValues() local
|
| D | quantize_model.cc | 633 RuntimeShape tensor_dims = in RequantizeConstant() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 514 uint32_t* tensor_dims = reinterpret_cast<uint32_t*>(tensor->dims->data); in ConvertTensorTypeToNNType() local
1533 const uint32_t* tensor_dims = reinterpret_cast<const uint32_t*>(dims->data); in AddNewInputConstantTensor() local
1849 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 | 3702 std::vector<int64_t> tensor_dims(dims.begin(), dims.end()); in ConvertPack() local
|
| D | convert_nodes_test.cc | 2915 std::vector<int> tensor_dims; in TEST_P() member
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1539 int tensor_dims[] = {1, input_size, 1}; in Create() local
|