| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| convolution_test.cc | 593 int64 output_feature; member in struct:xla::__anon39055::Convolve1DTestParam
607 int64 output_feature = GetParam().output_feature; local
614 output_feature};
648 std::vector<T> expect_elems(batch * output_feature * num_windows,
652 expected_r1->Reshape({batch, num_windows, output_feature})
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | [all...] |
| computation_builder.h | 364 // {output_feature, input_feature, height, width} = {0, 1, 2, 3}.
372 int64 input_second_spatial, int64 output_batch, int64 output_feature,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | 1007 llvm::Value* output_feature = index[dnums.output_feature_dimension()]; local
[all...] |
