HomeSort by relevance Sort by last modified time
    Searched refs:new_shape (Results 51 - 70 of 70) sorted by null

1 23

  /external/tensorflow/tensorflow/core/ops/
array_ops.cc     [all...]
  /external/tensorflow/tensorflow/python/data/experimental/ops/
batching.py 539 for original_shape, new_shape in zip(flat_original_shapes,
541 if not original_shape.is_compatible_with(new_shape):
  /external/tensorflow/tensorflow/python/ops/
ctc_ops.py     [all...]
math_ops.py     [all...]
array_ops.py     [all...]
  /external/tensorflow/tensorflow/python/ops/ragged/
ragged_tensor.py     [all...]
  /external/tensorflow/tensorflow/lite/delegates/nnapi/
nnapi_delegate_test.cc 674 std::initializer_list<int> new_shape) {
676 new_shape_ = AddConstInput<int>(TensorType_INT32, new_shape,
677 {static_cast<int>(new_shape.size())});
681 CreateReshapeOptions(builder_, builder_.CreateVector<int>(new_shape))
683 BuildInterpreter({input_shape, {static_cast<int>(new_shape.size())}});
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator.cc     [all...]
layout_assignment.cc 503 Shape new_shape = get_channel_constraints(instruction)
507 constraints->SetBufferLayout(new_shape.layout(), *buffer));
    [all...]
hlo_evaluator_typed_visitor.h     [all...]
  /external/tensorflow/tensorflow/python/ops/distributions/
util.py     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
literal.cc 581 Shape new_shape = shape();
582 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index);
585 Literal result(new_shape);
    [all...]
  /external/tensorflow/tensorflow/contrib/training/python/training/
sequence_queueing_state_saver.py     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
reshape_test.cc 106 auto new_shape = builder.GetShape(reshape).ConsumeValueOrDie(); local
    [all...]
  /external/tensorflow/tensorflow/python/keras/
backend.py     [all...]
  /external/tensorflow/tensorflow/lite/schema/
schema_generated.h 4494 std::vector<int32_t> new_shape; member in struct:tflite::ReshapeOptionsT
4504 const flatbuffers::Vector<int32_t> *new_shape() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
    [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
constant_folding.cc 1504 const NodeDef* new_shape = node_map_->GetNode(node.input(1)); local
    [all...]
  /external/tensorflow/tensorflow/python/client/
session_test.py     [all...]
  /external/tensorflow/tensorflow/c/
c_api.cc 1123 tensorflow::shape_inference::ShapeHandle new_shape = local
    [all...]
  /external/tensorflow/tensorflow/lite/toco/tflite/
operator.cc     [all...]

Completed in 881 milliseconds

1 23