OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:input_tf_shape
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h
282
auto
input_tf_shape
= GetTfShape(context_, src_index);
local
285
OP_REQUIRES(context_,
input_tf_shape
.dims() == 4,
287
input_tf_shape
.DebugString()));
289
GetOutputAndPadSizeInMklOrder(
input_tf_shape
, filter_tf_shape, strides,
380
TensorShape
input_tf_shape
= MakeInputTfShape(context, input_tensor);
variable
386
if (
input_tf_shape
.num_elements() == 0 ||
392
input_tf_shape
, filter_tf_shape, outbprop_tf_shape);
415
input_tf_shape
, filter_tf_shape, &fwd_input_dims, &fwd_filter_dims,
mkl_conv_grad_input_ops.cc
386
TensorShape
input_tf_shape
;
local
389
TensorShapeUtils::MakeShape(input_tensor.vec<int32>(), &
input_tf_shape
)
392
return
input_tf_shape
;
Completed in 108 milliseconds