Searched refs:data_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | bias_op_d9m_test.py | 39 data_dims = data_rank * (data_dim,)
41 shape = (batch_size,) + (channel_count,) + data_dims
43 shape = (batch_size,) + data_dims + (channel_count,)
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.cc | 329 ConvDimIndices GetDimIndices(const DataLayout& layout, const int data_dims) { in GetDimIndices() argument
333 dim_indices.data.depth_idx = data_dims - 1; in GetDimIndices()
334 dim_indices.data.batch_idx = data_dims - 2; in GetDimIndices()
339 dim_indices.data.depth_idx = data_dims - 2; in GetDimIndices()
340 dim_indices.data.batch_idx = data_dims - 1; in GetDimIndices()
345 dim_indices.data.depth_idx = data_dims - 1; in GetDimIndices()
365 ConvDimIndices GetDimIndices(const FilterLayout& layout, const int data_dims) { in GetDimIndices() argument
377 dim_indices.filter.input_idx = data_dims - 1; in GetDimIndices()
384 dim_indices.filter.output_idx = data_dims - 1; in GetDimIndices()
389 dim_indices.filter.input_idx = data_dims - 2; in GetDimIndices()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1638 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local
1639 Tensor indices = RandomReductionIndices(data_dims.size()); in TEST_F()
1642 .RandomInput(DT_BOOL, data_dims) in TEST_F()
1658 std::vector<int64_t> data_dims = RandomDims(); in TEST_F() local
1659 Tensor indices = RandomReductionIndices(data_dims.size()); in TEST_F()
1662 .RandomInput(DT_BOOL, data_dims) in TEST_F()
2332 std::vector<int64_t> data_dims = in TEST_F() local
2340 .RandomInput(type, data_dims) in TEST_F()
3198 std::vector<int64_t> data_dims = RandomDims(4, 4, 1, 8); in TEST_F() local
3200 std::uniform_int_distribution<int> radius(1, data_dims[3]); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 151 int64_t data_dims = tile_assignment.num_dimensions() - subgroup_types.size(); in Subgroup() local
152 std::vector<int64_t> perm(data_dims); in Subgroup()
174 if (tile_assignment.dim(i + data_dims) == 1) { in Subgroup()
180 dims.push_back(i + data_dims); in Subgroup()
211 .subspan(0, data_dims); in Subgroup()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBlock.h | 763 const Scalar* data, const DataDimensions& data_dims,
783 if (data_dims[dim] != desc.dimensions()[dim]) break;
812 TensorBlockIOSrc src(internal::strides<Layout>(Dimensions(data_dims)),
|