| /external/tensorflow/tensorflow/compiler/tests/ |
| depthwise_conv_op_test.py | 34 def ReferenceDepthwiseConv2D(input_tensor, filter_tensor, strides, padding,
39 in_channels = filter_tensor.shape[2]
49 filter_slice = filter_tensor[:, :, channel:channel+1, :]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| conv_ops_3d_test.py | 326 filter_tensor = constant_op.constant(
338 filter_tensor,
352 filter_tensor, filter_shape, conv, output_shape)
|
| depthwise_conv_op_test.py | 333 filter_tensor = constant_op.constant(
352 filter_tensor,
363 err = gradient_checker.compute_gradient_error(filter_tensor,
|
| conv_ops_test.py | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_ops.h | 365 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); variable
381 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor);
460 filter.SetUsrMem(fwd_filter_md, &filter_tensor);
519 const Tensor& filter_tensor) = 0;
|
| mkl_conv_ops.cc | 518 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); variable
584 filter.SetUsrMem(filter_md, &filter_tensor);
693 const memory::dims& filter_dims_tf_order, Tensor** filter_tensor) {
694 CHECK_NOTNULL(filter_tensor);
713 AllocateOutputSetMklShape(context, kOutputIndex_Filter, filter_tensor,
[all...] |
| mkl_conv_grad_filter_ops.cc | 463 const Tensor& filter_tensor) {
465 CHECK_EQ(TensorShapeUtils::IsVector(filter_tensor.shape()), true);
466 CHECK_EQ(TensorShapeUtils::MakeShape(filter_tensor.vec<int32>(),
|
| mkl_conv_grad_input_ops.cc | 396 const Tensor& filter_tensor) {
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 329 Tensor filter_tensor(
331 test::FillValues<float>(&filter_tensor, filter_values);
333 Const(root.WithOpName("filter_op"), Input::Initializer(filter_tensor));
[all...] |
