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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter_test.cc 107 ShapeInference::InferConvolveShape(activations->shape(),
135 ShapeInference::InferConvolveShape(activations->shape(),
276 ShapeInference::InferConvolveShape(
318 ShapeInference::InferConvolveShape(output->shape(), kernel->shape(),
350 ShapeInference::InferConvolveShape(output->shape(), kernel->shape(),
405 conv->shape(), ShapeInference::InferConvolveShape(
452 conv->shape(), ShapeInference::InferConvolveShape(
505 conv->shape(), ShapeInference::InferConvolveShape(
557 conv->shape(), ShapeInference::InferConvolveShape(
  /external/tensorflow/tensorflow/compiler/xla/service/
transpose_folding_test.cc 237 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape(
290 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape(
348 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape(
412 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape(
shape_inference.h 113 static StatusOr<Shape> InferConvolveShape(
shape_inference_test.cc 427 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
472 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
517 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
555 ShapeInference::InferConvolveShape(lhs_shape, rhs_shape, window, dnums);
    [all...]
hlo_verifier.cc 78 ShapeInference::InferConvolveShape(
    [all...]
shape_inference.cc     [all...]
hlo_evaluator.cc     [all...]
user_computation.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.cc 564 ShapeInference::InferConvolveShape(lhs_literal->shape(),

Completed in 231 milliseconds