| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| convolution_test.cc | 422 std::vector<int64> filter_dims = {2, 2, 2, 3, 3}; local
424 Shape filter_shape = ShapeUtil::MakeShape(F32, filter_dims);
459 auto filter_r5 = filter_r1->Reshape(filter_dims).ConsumeValueOrDie();
489 std::vector<int64> filter_dims = {3, 3, 5, 3}; local
491 Shape filter_shape = MakeShapeWrapper<T>(filter_dims);
523 auto filter_r4 = filter_r1->Reshape(filter_dims).ConsumeValueOrDie();
613 std::vector<int64> filter_dims = {window_size, input_feature, local
616 Shape filter_shape = MakeShapeWrapper<T>(filter_dims);
646 auto filter_r3 = filter_r1->Reshape(filter_dims).ConsumeValueOrDie();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_input_ops.cc | 113 const int64 filter_dims = mkl_context.filter_shape.GetDimension(); variable
116 filter_sizes, filter_dims, &filter_shape));
|
| mkl_conv_grad_filter_ops.cc | 185 mkl_context.filter_dims = 4;
229 mkl_context.grad_filter_shape.SetTfLayout(mkl_context.filter_dims,
275 int filter_dims; member in struct:tensorflow::MklConv2DCustomBackpropFilterOp::__anon39839
|
| mkl_conv_ops.h | 143 memory::dims* filter_dims) {
144 CHECK_NOTNULL(filter_dims);
178 *filter_dims = mkldnn_sizes;
187 memory::dims* filter_dims) {
188 CHECK_NOTNULL(filter_dims);
190 GetTfShape(context_, filter_index), filter_dims); local
302 memory::dims* input_dims, memory::dims* filter_dims,
307 CHECK_NOTNULL(filter_dims);
316 GetFilterSizeInMklOrder(input_shape, filter_shape, filter_dims);
|
| mkl_conv_ops.cc | 211 mkl_context.filter_dims = filter.dims();
302 mkl_filter_output_mkl_shape.SetTfDimOrder(mkl_context.filter_dims,
338 int filter_dims; member in struct:tensorflow::MklConv2DOp::__anon39841
361 CHECK_EQ(dnnLayoutCreate_F32(<_filter, filter_dims, filter_sizes,
531 memory::dims src_dims, filter_dims, padding_l, padding_r, strides; variable
539 src_tf_shape, filter_tf_shape, &src_dims, &filter_dims, &strides,
578 // Although filter shape (filter_dims) required is in MKL-DNN order,
582 : memory::desc(filter_dims, MklDnnType<T>(),
594 filter.SetOpMemDesc(filter_dims, memory::format::any);
[all...] |
