HomeSort by relevance Sort by last modified time
    Searched refs:GetTfShape (Results 1 - 12 of 12) sorted by null

  /external/tensorflow/tensorflow/core/util/
mkl_util_test.cc 39 TensorShape a_mkldnn_tf_shape = a_mkldnn_shape.GetTfShape();
51 TensorShape b_mkldnn_tf_shape = b_mkldnn_shape.GetTfShape();
mkl_util.h 400 return this->GetTfShape() == input_shape.GetTfShape() &&
415 return this->GetTfShape() == input_shape;
477 inline TensorShape GetTfShape() const {
786 inline TensorShape GetTfShape(OpKernelContext* context, size_t input_idx) {
794 return input_mkl_shape.GetTfShape();
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc 64 ? src_mkl_shape.GetTfShape()
mkl_conv_ops.h 189 GetFilterSizeInMklOrder(GetTfShape(context_, src_index),
190 GetTfShape(context_, filter_index), filter_dims);
282 auto input_tf_shape = GetTfShape(context_, src_index);
283 auto filter_tf_shape = GetTfShape(context_, filter_index);
382 TensorShape outbprop_tf_shape = GetTfShape(context, kOutbpropIdx);
mkl_tfconv_op.h 102 TensorShape output_shape = input_shape.GetTfShape();
mkl_reshape_op.cc 170 ? mkl_shape_input.GetTfShape().num_elements()
221 TensorShape shape_from = mkl_shape_input.GetTfShape();
mkl_concat_op.cc 609 ? input_shapes[0].GetTfShape()
622 s.IsMklTensor() ? s.GetTfShape() : input_tensors[i].shape();
667 s.IsMklTensor() ? s.GetTfShape() : input_tensors[i].shape();
677 dst_dims = TFShapeToMklDnnDims(input_shapes[0].GetTfShape());
693 ? TFShapeToMklDnnDims(input_shapes[k].GetTfShape())
    [all...]
mkl_fused_batch_norm_op.cc     [all...]
mkl_aggregate_ops.cc 320 src1_shape = input1_in_mkl_format ? src1_mkl_shape.GetTfShape()
322 src2_shape = input2_in_mkl_format ? src2_mkl_shape.GetTfShape()
mkl_conv_grad_filter_ops.cc 459 return GetTfShape(context, input_idx);
512 TensorShape obp_tf_shape = GetTfShape(context, 2);
mkl_conv_ops.cc 536 auto src_tf_shape = GetTfShape(context, kInputIndex_Src);
537 auto filter_tf_shape = GetTfShape(context, kInputIndex_Filter);
    [all...]
mkl_conv_grad_input_ops.cc 397 return GetTfShape(context, kInputIndex_Filter);

Completed in 966 milliseconds