Searched refs:inner_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_op_test.py | 140 inner_dims = len(shape) - axis - 1
145 (slice(None),) * inner_dims)
147 (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 | 262 inner_dims = array_ops.concat(
265 return RaggedTensorDynamicShape([], inner_dims)
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1953 absl::flat_hash_set<int64> inner_dims; in HandleSlice() local
1958 inner_dims.insert(dim); in HandleSlice()
1961 const bool is_trivial_copy = (inner_dims.size() == num_dims); in HandleSlice()
1973 [&inner_dims](int64 dim) { return inner_dims.contains(dim); }, in HandleSlice()
1981 const int64 memcpy_dim = LayoutUtil::Minor(layout, inner_dims.size()); in HandleSlice()
1994 for (int64 i = 0; i < num_dims - inner_dims.size() - 1; ++i) { in HandleSlice()
2410 std::vector<int64> inner_dims(output_min2maj.begin(), concat_dim_layout_itr); in EmitFastConcatenate() local
2444 std::accumulate(inner_dims.begin(), inner_dims.end(), 1l, in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 447 inner_dims = array_ops.size(inner_shape)
451 outer_dims + 1 + inner_dims)
|