Home
last modified time | relevance | path

Searched refs:filter_tensor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_input_ops.cc318 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
334 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor); in Compute()
442 filter.SetUsrMem(fwd_filter_md, &filter_tensor); in Compute()
447 static_cast<T*>(const_cast<T*>(filter_tensor.flat<T>().data())); in Compute()
508 const Tensor& filter_tensor) { in MakeFilterTfShape() argument
Dmkl_conv_grad_filter_ops.cc381 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); in Compute() local
397 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor); in Compute()
628 const Tensor& filter_tensor) { in MakeFilterTfShape() argument
630 CHECK_EQ(TensorShapeUtils::IsVector(filter_tensor.shape()), true); in MakeFilterTfShape()
631 CHECK_EQ(TensorShapeUtils::MakeShape(filter_tensor.vec<int32>(), in MakeFilterTfShape()
Dmkl_conv_ops.cc924 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); in Compute() local
1048 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
1120 CacheFilter(context, conv_fwd_pd, filter_data, filter_tensor, in Compute()
1128 filter.SetUsrMem(filter_md, &filter_tensor); in Compute()
1142 const_cast<Tfilter*>(filter_tensor.flat<Tfilter>().data())); in Compute()
1307 Tensor** filter_tensor) { in AllocatePersistentTensor() argument
1308 DCHECK(filter_tensor); in AllocatePersistentTensor()
1314 &cached_filter_data_ptensor_, filter_tensor)); in AllocatePersistentTensor()
1331 Tensor** filter_tensor) { in AllocateFilterOutputTensor() argument
1332 CHECK_NOTNULL(filter_tensor); in AllocateFilterOutputTensor()
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Ddepthwise_conv_op_test.py34 def ReferenceDepthwiseConv2D(input_tensor, filter_tensor, strides, padding, argument
39 in_channels = filter_tensor.shape[2]
49 filter_slice = filter_tensor[:, :, channel:channel+1, :]
/external/tensorflow/tensorflow/python/kernel_tests/
Ddepthwise_conv_op_test.py380 filter_tensor = constant_op.constant(
404 filter_tensor,
418 filter_tensor, filter_shape, depthwise_conv2d, output_shape)
Dconv_ops_3d_test.py419 filter_tensor = constant_op.constant(
431 filter_tensor,
447 filter_tensor, filter_shape, conv, output_shape)
Dconv_ops_test.py390 filter_tensor = self._CreateNumpyTensor(filter_in_sizes)
395 filter_tensor, [1] + list(strides) + [1],
1674 filter_tensor = constant_op.constant(
1687 filter_tensor,
1701 jacob_t, jacob_n = gradient_checker.compute_gradient(filter_tensor,
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples.py1709 filter_tensor = tf.constant(
1720 filter_tensor,
1727 filter_tensor,
1781 filter_tensor = tf.constant(
1789 filter_tensor,
1796 filter_tensor,
3663 filter_tensor = tf.placeholder(
3668 filter_tensor,
3673 input_tensors = [input_tensor, filter_tensor]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc329 Tensor filter_tensor( in TestQuantizeConv2D() local
331 test::FillValues<float>(&filter_tensor, filter_values); in TestQuantizeConv2D()
333 Const(root.WithOpName("filter_op"), Input::Initializer(filter_tensor)); in TestQuantizeConv2D()