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

  /external/tensorflow/tensorflow/core/kernels/
conditional_accumulator.h 67 Status ValidateShape(const Tensor* tensor)
123 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor));
sparse_conditional_accumulator.h 84 Status ValidateShape(
399 OP_REQUIRES_OK_BOOLEAN(ctx, this->ValidateShape(*tensor, has_known_shape));
  /external/tensorflow/tensorflow/compiler/xla/
shape_layout.cc 67 TF_CHECK_OK(ShapeUtil::ValidateShape(shape_));
shape_util.cc 139 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape));
212 TF_DCHECK_OK(ShapeUtil::ValidateShape(shape));
235 TF_DCHECK_OK(ValidateShape(*shape));
266 TF_DCHECK_OK(ValidateShape(*shape));
591 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(result));
702 TF_DCHECK_OK(ValidateShape(shape));
716 TF_DCHECK_OK(ValidateShape(shape));
723 TF_DCHECK_OK(ValidateShape(shape));
748 TF_DCHECK_OK(ValidateShape(shape));
798 return ValidateShape(shape)
    [all...]
shape_util.h 352 static Status ValidateShape(const Shape& shape);
  /external/tensorflow/tensorflow/compiler/xla/service/
local_service.cc 93 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(argument_shape));
transfer_manager.cc 191 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(on_host_shape));
service.cc 232 return ShapeUtil::ValidateShape(shape_with_layout);
    [all...]
hlo_evaluator.cc     [all...]
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/core/graph/
graph_constructor.cc 190 Status ValidateShape(Node* node);
558 Status GraphConstructor::ValidateShape(Node* node) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc 91 TF_CHECK_OK(ShapeUtil::ValidateShape(shape));
llvm_util.cc 247 TF_RETURN_IF_ERROR(ShapeUtil::ValidateShape(shape));

Completed in 142 milliseconds