HomeSort by relevance Sort by last modified time
    Searched defs:filter_tensor (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_filter_ops.cc 381 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); local
397 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor);
628 const Tensor& filter_tensor) {
630 CHECK_EQ(TensorShapeUtils::IsVector(filter_tensor.shape()), true);
631 CHECK_EQ(TensorShapeUtils::MakeShape(filter_tensor.vec<int32>(),
mkl_conv_grad_input_ops.cc 318 const Tensor& filter_tensor = MklGetInput(context, kFilterIdx); local
334 TensorShape filter_tf_shape = MakeFilterTfShape(context, filter_tensor);
442 filter.SetUsrMem(fwd_filter_md, &filter_tensor);
447 static_cast<T*>(const_cast<T*>(filter_tensor.flat<T>().data()));
508 const Tensor& filter_tensor) {
mkl_conv_ops.cc 924 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter); variable
    [all...]

Completed in 183 milliseconds