Searched refs:inner_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | gather_op_test.py | 161 inner_dims = len(shape) - axis - 1
166 (slice(None),) * inner_dims)
168 (slice(None),) * inner_dims)
230 inner_dims = len(shape) - axis - 1
235 (slice(None),) * inner_dims)
237 (slice(None),) * inner_dims)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fft_test.py | 62 def _VerifyFftMethod(self, inner_dims, complex_to_input, input_to_expected, argument
64 for indims in inner_dims:
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 288 inner_dims = array_ops.concat(
292 return RaggedTensorDynamicShape([], inner_dims)
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1867 absl::flat_hash_set<int64> inner_dims; in HandleSlice() local
1872 inner_dims.insert(dim); in HandleSlice()
1875 const bool is_trivial_copy = (inner_dims.size() == num_dims); in HandleSlice()
1887 [&inner_dims](int64 dim) { return inner_dims.contains(dim); }, in HandleSlice()
1895 const int64 memcpy_dim = LayoutUtil::Minor(layout, inner_dims.size()); in HandleSlice()
1908 for (int64 i = 0; i < num_dims - inner_dims.size() - 1; ++i) { in HandleSlice()
2471 std::vector<int64> inner_dims(output_min2maj.begin(), concat_dim_layout_itr); in EmitFastConcatenate() local
2505 std::accumulate(inner_dims.begin(), inner_dims.end(), 1l, in EmitFastConcatenate()
|