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

  /external/tensorflow/tensorflow/core/kernels/
mkl_concat_op.cc 800 TensorShape tf_shape_output; local
801 tf_shape_output.AddDim(dnn_shape_output.GetSerializeBufferSize());
803 tf_shape_output, &output_tensor);
    [all...]
mkl_fused_batch_norm_op.cc 129 TensorShape tf_shape_output; variable
139 tf_shape_output.AddDim(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
142 AllocateOutputSetMklShape(context, 0, &output, tf_shape_output,
    [all...]

Completed in 70 milliseconds